# -*- coding:utf-8 -*-

import argparse

def parsers():
    parser = argparse.ArgumentParser(description="Bert model of argparse")
    parser.add_argument("--bert_pred", type=str, default="./Bert-base-chinese", help="bert 预训练模型")
    parser.add_argument("--class_num", type=int, default=5, help="分类数")
    parser.add_argument("--max_len", type=int, default=512, help="句子的最大长度")
    parser.add_argument("--batch_size", type=int, default=8, help="迭代批次")
    parser.add_argument("--epochs", type=int, default=2, help="迭代次数")
    parser.add_argument("--dropout", type=float, default=0.5, help="失活率")
    parser.add_argument("--filter_sizes", type=list, default=[5, 3, 4], help="TextCnn 的卷积核大小")
    parser.add_argument("--num_filters", type=int, default=256, help="TextCnn 的卷积输出")
    parser.add_argument("--encode_layer", type=int, default=12, help="bert-encoder层数")
    parser.add_argument("--hidden_size", type=int, default=768, help="bert 层输出维度")
    parser.add_argument("--save_model", type=str, default='./Bert-TextCNN-model', help="模型保存路径")

    args = parser.parse_args()
    return args
